from core.train import model_train
from sklearn.datasets import load_iris
from utils.xlsx import xlsx_add_data,create_xlsx_data
if __name__ == '__main__':
    CLASSIFIER_MODEL = ["random_forest","naive_bayes"]
    XLSX_HEADER = ["model_name","classifier_name","test_size","random_state","accuracy","precision","recall","f1","mcc","balance"]
    XLSX_FILE = "./results/record.xlsx"
    create_xlsx_data(file_path=XLSX_FILE,xlsx_header=XLSX_HEADER)
    data,target = load_iris(return_X_y=True)
    model_train(
        classifier_model_list=CLASSIFIER_MODEL,
        data=data,
        target=target,
        record_fn=xlsx_add_data,
        record_file_path=XLSX_FILE,
        epoch=10
    )